import numpy as np

def gen_golden_data_thresholded_relu():
    # 生成随机输入数据（包含负值和正值，范围 -5~10，形状 [8, 2048]）
    input_x = np.random.uniform(-5, 10, [8, 2048]).astype(np.float32)
    
    # 设置阈值（通常为1.0）
    threshold = 1.0
    
    # 计算 thresholded ReLU: f(x) = x if x > threshold else 0
    golden = np.where(input_x > threshold, input_x, 0).astype(np.float32)
    
    # 保存 tiling 信息（保持原逻辑）
    tiling = np.array([16384, 8], dtype=np.uint32)
    tiling.tofile("./input/input_tiling.bin")
    
    # 保存输入和输出数据
    input_x.tofile("./input/input_x.bin")
    golden.tofile("./output/golden.bin")

if __name__ == "__main__":
    gen_golden_data_thresholded_relu()